%run train-cGAN
# %run unit_test

self.device is cuda:0
64
Epoch: 0, Itration: 0
loss_G_GAN = 5.2222394943237305, loss_G_L1 = 0.42152929306030273 * 5.0
loss_D = 0.6994704008102417

Epoch: 76, Itration: 1000
loss_G_GAN = 1.7559757232666016, loss_G_L1 = 0.05914055183529854 * 5.0
loss_D = 0.32040828466415405

Epoch: 153, Itration: 2000
loss_G_GAN = 2.0574169158935547, loss_G_L1 = 0.05481714755296707 * 5.0
loss_D = 0.2651404142379761

save model to ./checkpoints/pix2food.pkl at iteration = 3000
Epoch: 230, Itration: 3000
loss_G_GAN = 1.2604695558547974, loss_G_L1 = 0.04347262904047966 * 5.0
loss_D = 0.3395516872406006

Finally save model to ./checkpoints/pix2food.pkl



 # --- Validation Set --- # 

# %run train-cGAN

# %run train